package com.ants.boot.utils.http;

import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;

/**
 * @author 蚂蚁会花呗
 * @date 2022/10/28 16:17
 * HttpURLConnection 从地址下载文件
 */
public class HttpUrlConUtils {

    private static Logger logger = LoggerFactory.getLogger(HttpUrlConUtils.class);


    /**
     * 从远程地址下载文件，保存到本地
     * @param url 文件地址
     * @param filePath 需要保存到服务器的路径必须以/结尾
     * @param filename 文件名称
     * @return
     */
    public static String downloadForUrl(String url, String filePath, String filename){
        if (StringUtils.isBlank(filePath) || StringUtils.isBlank(filename)){
            logger.error("filePath: {}, filename: {}",filePath,filename);
            throw new RuntimeException("路径或者文件名称不能为空！");
        }
        if (!filePath.endsWith("/")){
            filePath+="/";
        }
        return download(url,filePath,filename);
    }

    /**
     * 从服务器get请求下载文件并保存自定义格式
     * @param url
     * @param filePath
     * @param filename
     * @return
     */
    private static String download(String url, String filePath, String filename){
        //创建不同的文件夹目录
        File file=new File(filePath);
        //判断文件夹是否存在
        if (!file.exists()) {
            //如果文件夹不存在，则创建新的的文件夹
            file.mkdirs();
        }
        FileOutputStream fileOut = null;
        HttpURLConnection conn = null;
        InputStream inputStream = null;
        try {
            // 建立链接
            URL httpUrl=new URL(url);
            conn=(HttpURLConnection) httpUrl.openConnection();
            //以Post方式提交表单，默认get方式
            conn.setRequestMethod("GET");
            conn.setDoInput(true);
            conn.setDoOutput(true);
            // post方式不能使用缓存
            conn.setUseCaches(false);
            //连接指定的资源
            conn.connect();
            //获取网络输入流
            inputStream=conn.getInputStream();
            BufferedInputStream bis = new BufferedInputStream(inputStream);
            //写入到文件（注意文件保存路径的后面一定要加上文件的名称）
            fileOut = new FileOutputStream(filePath + filename);
            BufferedOutputStream bos = new BufferedOutputStream(fileOut);
            byte[] buf = new byte[4096];
            int length = bis.read(buf);
            //保存文件
            while(length != -1) {
                bos.write(buf, 0, length);
                length = bis.read(buf);
            }
            bos.close();
            bis.close();
            conn.disconnect();
        } catch (Exception e) {
            logger.error("download file failed！");
            e.printStackTrace();
        }
        return filePath+filename;
    }

}
